package file

import (
	"io/ioutil"
	"os"
	"path"
	"strings"
)

func SearchPic(dir string, imgSuffix string) (int, []string) {

	picFileCount := 0
	picFiles := make([]string, 0)

	files, _ := ioutil.ReadDir(dir) // 文件路径
	for _, f := range files {
		suffix := path.Ext(f.Name())
		if ok := strings.Contains(imgSuffix, suffix); ok && suffix != "" {
			picFileCount += 1
			picFiles = append(picFiles, f.Name())
		}
	}

	return picFileCount, picFiles
}

/**
 * 写文件
 * @param {[type]} name    [description]
 * @param {[type]} content string        [description]
 */
func WriteWithIoutil(name string, content string) error {
	var f *os.File
	if CheckFileIsExist(name) { // 如果文件存在
		f, _ = os.OpenFile(name, os.O_WRONLY|os.O_APPEND, 0666) //打开文件
	} else {
		f, _ = os.Create(name) // 创建文件
		os.Chmod(name, 0666)
	}
	_, err := f.Write([]byte(content))
	f.Close()

	return err
}

/**
 * 判断文件是否存在
 */
func CheckFileIsExist(filename string) bool {
	exist := true
	if _, err := os.Stat(filename); os.IsNotExist(err) {
		exist = false
	}
	return exist
}
